package code1.binary_tree;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * 原题链接：https://leetcode.cn/problems/binary-tree-right-side-view/
 */
public class P199 {

    //二叉树的右视图
    //思路，层序遍历，把每一层的最右边的元素加入结果集
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> result=new ArrayList<>();
        Queue<TreeNode> queue=new LinkedList<>();
        if(root==null){
            return result;
        }
        queue.add(root);
        while(!queue.isEmpty()){
            int size=queue.size();
            for(int i=0;i<size;i++){
                TreeNode node = queue.remove();
                if(i==size-1){
                    result.add(node.val);       //添加最右端元素
                }
                if(node.left!=null){
                    queue.add(node.left);       //添加左节点
                }
                if(node.right!=null){
                    queue.add(node.right);      //添加右节点
                }
            }
        }
        return result;
    }
}
